package club.xiaojiawei.graph;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 6/23/22 9:35 PM
 * @question 841. 钥匙和房间
 * @description 有 n 个房间，房间按从 0 到 n - 1 编号。最初，除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而，你不能在没有获得钥匙的时候进入锁住的房间。
 * 当你进入一个房间，你可能会在里面找到一套不同的钥匙，每把钥匙上都有对应的房间号，即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。
 * 给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true，否则返回 false。
 */
public class CanVisitAllRooms841 {

    public static void main(String[] args) {
        CanVisitAllRooms841 test = new CanVisitAllRooms841();
        boolean result = test.canVisitAllRooms2(new ArrayList<>() {
            {
                add(Arrays.asList(1, 3));
                add(Arrays.asList(3, 0, 1));
                add(List.of(2));
                add(List.of(0));
            }
        });
        System.out.println(result);
    }

    /**
     * bfs
     * 时间复杂度：O(n+m)
     * 空间复杂度：O(n)
     * @param rooms
     * @return
     */
    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        boolean[] flag = new boolean[rooms.size()];
        LinkedList<Integer> queue = new LinkedList<>(rooms.get(0));
        flag[0] = true;
        int count = 1;
        while (!queue.isEmpty()){
            Integer pop = queue.pop();
            if (flag[pop]){
                continue;
            }
            queue.addAll(rooms.get(pop));
            flag[pop] = true;
            count++;
        }
        return count == rooms.size();
    }

    /**
     * dfs
     * 时间复杂度：O(n+m)
     * 空间复杂度：O(n)
     * @param rooms
     * @return
     */
    public boolean canVisitAllRooms2(List<List<Integer>> rooms) {
        flag = new boolean[rooms.size()];
        reversion(rooms, 0);
        return count == rooms.size();
    }

    boolean[] flag;

    int count;

    public void reversion(List<List<Integer>> rooms, int n){
        flag[n] = true;
        count++;
        List<Integer> integers = rooms.get(n);
        for (Integer integer : integers) {
            if (!flag[integer]){
                reversion(rooms, integer);
                if (count == rooms.size()){
                    return;
                }
            }
        }
    }

}
